#include "common.h"

// 声明测试函数
void test_tensor_operations();
void test_autograd_examples();
void test_neural_network();
void test_custom_operators();

int main() {
  std::cout << "PyTorch C++ API Study Framework" << std::endl;
  std::cout << "PyTorch Version: " << TORCH_VERSION << std::endl;

  // 依次运行所有测试
  std::cout << "\n=== Running Tensor Operations Test ===" << std::endl;
  test_tensor_operations();

  std::cout << "\n=== Running Autograd Examples Test ===" << std::endl;
  test_autograd_examples();

  std::cout << "\n=== Running Neural Network Test ===" << std::endl;
  test_neural_network();

  std::cout << "\n=== Running Custom Operators Test ===" << std::endl;
  test_custom_operators();

  std::cout << "\nAll tests completed!" << std::endl;
  return 0;
}